# -*- coding: utf-8 *-*

#   This file is part of Sql-Rice.
#
#    Sql-Rice is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 3 of the License, or
#    (at your option) any later version.
#
#    Sql-Rice is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with Sql-Rice; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA

import unittest
import tools.ruledit as ruledit
import rules.models as models


class TestRuleEditor(unittest.TestCase):

    def setUp(self):
        models.metadata.bind = 'sqlite:///:memory:'
        models.setup_all(True)
        self.rut = models.Validator(name='rut')
        models.session.commit()

    def test_add_rule(self):
        editor = ruledit.RuleEditor()
        editor.do_field('rut add rut')
        rule = models.Rule.get_by(pattern='rut')
        self.assertIsNotNone(rule)
        self.assertIn(self.rut, rule.validators)
